from .mobilenetv2 import MobileNet
from .dla import get_pose_net
import torchvision.models as models

backone = {
    'mobilenetv2': MobileNet,
    'resnet50': models.resnet50, 
    'dla34': get_pose_net,
}

def get_backone(name, **kwargs):
    return backone[name](**kwargs)

